package com.eva.config.quartz;

import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import com.eva.core.constants.Constants;
import com.eva.core.job.SchedulerProxy;
import com.eva.dao.system.SystemJobMapper;
import com.eva.dao.system.SystemJobSnippetMapper;
import com.eva.dao.system.model.SystemJob;
import com.eva.dao.system.model.SystemJobSnippet;
import com.eva.service.system.SystemJobService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;
import java.util.List;

@Slf4j
@Component
public class JobInitializer {

    @Autowired
    private SystemJobService systemJobService;

    @Autowired
    private SystemJobMapper systemJobMapper;

    @Autowired
    private SystemJobSnippetMapper systemJobSnippetMapper;

    @Autowired
    private SchedulerProxy schedulerProxy;

    /**
     * 初始化Jobs
     */
    @PostConstruct
    public void initJobs () {
        log.info("Initializing jobs");
        long startTime = System.currentTimeMillis();
        SystemJob queryJobDto = new SystemJob();
        queryJobDto.setStatus(Constants.Job.JobStatus.PAUSE.getCode());
        queryJobDto.setDeleted(Boolean.FALSE);
        List<SystemJob> jobs = systemJobService.findList(queryJobDto);
        for (SystemJob job : jobs) {
            schedulerProxy.createJob(new SchedulerProxy.Job(job.getId(), job.getHandler(), job.getCronExpression(), job.getDisallowConcurrent()));
            log.info("job '{}' initialized.", job.getJobName());
        }
        log.info("Jobs initialization completed in {} ms", System.currentTimeMillis() - startTime);
    }

    /**
     * 修复Jobs和Snippets，防止因服务器宕机导致正在执行中的Job或Snippet状态未得到处理，导致任务或分片丢失
     * 是否需要修复？
     * 场景1：处理数据时未做事务，1000条数据处理了500条后服务器宕机，如果任务重新执行，会处理1000条记录。（错误，这会要求任务业务需要实现幂等）
     * 场景2：处理数据时添加了事务，1000条数据处理了500条后服务器宕机，任务重新执行，会处理1000条记录。（正确）
     */
    @PostConstruct
    public void repairJobs () {
        log.info("Repair jobs");
        // 将运行中的JOB调整为准备状态 & 将预计下一次执行时间调整为null
        systemJobMapper.update(null, new LambdaUpdateWrapper<SystemJob>()
                .set(SystemJob::getStatus, Constants.Job.JobStatus.READY.getCode())
                .set(SystemJob::getNextFireTime, null)
                .eq(SystemJob::getStatus, Constants.Job.JobStatus.RUNNING.getCode())
                .eq(SystemJob::getDeleted, Boolean.FALSE)
        );
        log.info("Repair job snippets");
        // 将运行中的SNIPPET调整为准备状态
        SystemJobSnippet newSnippet = new SystemJobSnippet();
        newSnippet.setStatus(Constants.Job.SnippetStatus.READY.getCode());
        systemJobSnippetMapper.update(newSnippet, new LambdaUpdateWrapper<SystemJobSnippet>()
                .eq(SystemJobSnippet::getStatus, Constants.Job.SnippetStatus.READY.getCode())
        );
    }
}
